/*!	 valuedescskeletonlink.cpp
**	 Template File
**
**	......... ... 2013 Ivan Mahonin
**
**	This package is free software; you can redistribute it and/or
**	modify it under the terms of the GNU General Public License as
**	published by the Free Software Foundation; either version 2 of
**	the License, or (at your option) any later version.
**
**	This package is distributed in the hope that it will be useful,
**	but WITHOUT ANY WARRANTY; without even the implied warranty of
**	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
**	General Public License for more details.
**
*/

#ifdef USING_PCH
#	include "pch.h"
#else
#ifdef HAVE_CONFIG_H
#	include <config.h>
#endif

#include <synfig/general.h>

#include "layerparamconnect.h"
#include "valuenodelinkconnect.h"
#include "valuenodereplace.h"
#include "valuedescskeletonlink.h"
#include "valuenodestaticlistinsert.h"

#include <synfigapp/canvasinterface.h>
#include <synfig/boneweightpair.h>
#include <synfig/valuenodes/valuenode_const.h>
#include <synfig/valuenodes/valuenode_composite.h>
#include <synfig/valuenodes/valuenode_bone.h>
#include <synfig/valuenodes/valuenode_bonelink.h>
#include <synfig/valuenodes/valuenode_boneweightpair.h>
#include <synfig/valuenodes/valuenode_staticlist.h>
#include <synfig/valuenodes/valuenode_weightedaverage.h>
#include <synfig/valueoperations.h>
#include <synfig/weightedvalue.h>

#include <synfigapp/localization.h>

#endif

using namespace std;
using namespace etl;
using namespace synfig;
using namespace synfigapp;
using namespace Action;

ACTION_INIT(Action::ValueDescSkeletonLink);
ACTION_SET_NAME(Action::ValueDescSkeletonLink, "ValueDescSkeletonLink");
ACTION_SET_LOCAL_NAME(Action::ValueDescSkeletonLink, N_("Link to Skeleton"));
ACTION_SET_TASK(Action::ValueDescSkeletonLink, "bone_link");
ACTION_SET_CATEGORY(Action::ValueDescSkeletonLink, Action::CATEGORY_VALUEDESC);
ACTION_SET_PRIORITY(Action::ValueDescSkeletonLink, 0);
ACTION_SET_VERSION(Action::ValueDescSkeletonLink, "0.0");
ACTION_SET_CVS_ID(Action::ValueDescSkeletonLink, "$Id$");

Action::ValueDescSkeletonLink::ValueDescSkeletonLink():
    time(0)
{
}

Action::ParamVocab
Action::ValueDescSkeletonLink::get_param_vocab()
{
    ParamVocab ret(Action::CanvasSpecific::get_param_vocab());

    ret.push_back(ParamDesc("selected_value_desc", Param::TYPE_VALUEDESC)
                  .set_local_name(_("ValueDesc to link"))
                  .set_supports_multiple()
                 );
    ret.push_back(ParamDesc("value_desc", Param::TYPE_VALUEDESC)
                  .set_local_name(_("ValueDesc of Skeleton's Bone"))
                 );
    ret.push_back(ParamDesc("time", Param::TYPE_TIME)
                  .set_local_name(_("Time"))
                  .set_optional()
                 );

    return ret;
}

bool
Action::ValueDescSkeletonLink::is_candidate(const ParamList &x)
{
    ParamList::const_iterator i;

    ValueDesc value_desc(x.find("value_desc")->second.get_value_desc());

    if (!candidate_check(get_param_vocab(), x)) {
        return false;
    }

    return value_desc.parent_is_value_node()
           && ValueNode_Bone::Handle::cast_dynamic(value_desc.get_parent_value_node());
}

bool
Action::ValueDescSkeletonLink::set_param(const synfig::String& name, const Action::Param &param)
{
    if (name == "value_desc" && param.get_type() == Param::TYPE_VALUEDESC
            && param.get_value_desc().parent_is_value_node()
            && ValueNode_Bone::Handle::cast_dynamic(param.get_value_desc().get_parent_value_node())
            && param.get_value_desc().get_parent_desc().parent_is_value_node()
            && ValueNode_StaticList::Handle::cast_dynamic(param.get_value_desc().get_parent_desc().get_parent_value_node())) {
        value_desc = param.get_value_desc();
        return true;
    }

    if (name == "selected_value_desc" && param.get_type() == Param::TYPE_VALUEDESC) {
        value_desc_list.push_back(param.get_value_desc());
        return true;
    }

    if (name == "time" && param.get_type() == Param::TYPE_TIME) {
        time = param.get_time();
        return true;
    }

    return Action::CanvasSpecific::set_param(name, param);
}

bool
Action::ValueDescSkeletonLink::is_ready()const
{
    if (value_desc_list.empty()) {
        return false;
    }

    if (!value_desc) {
        return false;
    }

    return Action::CanvasSpecific::is_ready();
}

void
Action::ValueDescSkeletonLink::prepare()
{
    if (value_desc_list.empty()) {
        throw Error(Error::TYPE_NOTREADY);
    }

    clear();

    // get bone
    ValueNode_Bone::Handle bone_value_node;

    if (value_desc.parent_is_value_node()) {
        bone_value_node = ValueNode_Bone::Handle::cast_dynamic(value_desc.get_parent_value_node());
    }

    if (!bone_value_node) {
        throw Error(Error::TYPE_NOTREADY);
    }

    // get static list of bones (skeleton layer)
    const ValueDesc &parent = value_desc.get_parent_desc();
    ValueNode_StaticList::Handle bone_list_value_node;

    if (parent.parent_is_value_node()) {
        bone_list_value_node = ValueNode_StaticList::Handle::cast_dynamic(parent.get_parent_value_node());
    }

    if (!bone_list_value_node) {
        throw Error(Error::TYPE_NOTREADY);
    }

    // bones list
    typedef std::set<ValueNode_Bone::Handle> Set;
    ValueBase::List value_bone_list = (*bone_list_value_node)(time).get_list();
    Set list;

    for (int i = 0; i < bone_list_value_node->link_count(); ++i) {
        ValueNode_Bone::Handle bone_value_node =
            ValueNode_Bone::Handle::cast_dynamic(bone_list_value_node->get_link(i));

        if (bone_value_node) {
            list.insert(bone_value_node);
        }
    }

    if (list.empty()) {
        throw Error(Error::TYPE_NOTREADY);
    }

    // process all selected ducks
    Set current_list;

    for (std::list<ValueDesc>::iterator iter = value_desc_list.begin(); iter != value_desc_list.end(); ++iter) {
        ValueDesc& value_desc(*iter);

        // skip region/outline origin
        if (value_desc.parent_is_layer()
                && value_desc.get_param_name() == "origin"
                && (value_desc.get_layer()->get_name() == "advanced_outline"
                    || value_desc.get_layer()->get_name() == "outline"
                    || value_desc.get_layer()->get_name() == "region")) {
            continue;
        }

        // check type
        Type &type(value_desc.get_value_type());

        if (!ValueNode_BoneLink::check_type(type)
                || !ValueNode_WeightedAverage::check_type(type)
                || !ValueVector::check_type(type)) {
            continue;
        }

        // don't link bones to bones
        if (value_desc.parent_is_value_node()
                && ValueNode_Bone::Handle::cast_dynamic(value_desc.get_parent_value_node())) {
            continue;
        }

        // List of bones influencing current item
        current_list.clear();

        for (Set::iterator i = list.begin(); i != list.end(); ++i)
            if ((*i)->have_influence_on(time, ValueVector::get_vector(value_desc.get_value(time)))) {
                current_list.insert(*i);
            }

        if (current_list.empty()) {
            continue;
        }

        ValueNode::Handle node;

        if (current_list.size() > 1) {
            // make average node
            ValueNode_WeightedAverage::Handle average_node = new ValueNode_WeightedAverage(type, get_canvas());

            // get type of weighted value
            types_namespace::TypeWeightedValueBase *wt = ValueAverage::get_weighted_type_for(type);
            assert(wt != NULL);

            // add each bone from influence_list to Average convert
            for (Set::iterator i = current_list.begin(); i != current_list.end(); ++i) {
                // make bone link
                ValueNode_BoneLink::Handle bone_link_node =
                    ValueNode_BoneLink::create(value_desc.get_value(time));

                bone_link_node->set_link("bone", ValueNode_Const::create(ValueBase(*i), get_canvas()));
                bone_link_node->set_link("base_value",
                                         ValueNode_Composite::create(
                                             ValueTransformation::back_transform(
                                                 bone_link_node->get_bone_transformation(time),
                                                 value_desc.get_value(time))));

                // make weighted value
                ValueNode_Composite::Handle weighted_node =
                    ValueNode_Composite::create(wt->create_weighted_value(1, value_desc.get_value(time)), get_canvas());

                weighted_node->set_link("value", bone_link_node);

                // add
                average_node->add(ValueNode::Handle(weighted_node));
            }

            node = average_node;
        } else {
            // make bone link
            ValueNode_BoneLink::Handle bone_link_node =
                ValueNode_BoneLink::create(value_desc.get_value(time));

            bone_link_node->set_link("bone", ValueNode_Const::create(ValueBase(*current_list.begin()), get_canvas()));
            bone_link_node->set_link("base_value",
                                     ValueNode_Composite::create(
                                         ValueTransformation::back_transform(
                                             bone_link_node->get_bone_transformation(time),
                                             value_desc.get_value(time))));

            node = bone_link_node;
        }

        if (!node) {
            continue;
        }

        // enqueue suitable action to assign node
        if (value_desc.parent_is_canvas()) {
            Action::Handle action = ValueNodeReplace::create();
            action->set_param("canvas", get_canvas());
            action->set_param("canvas_interface", get_canvas_interface());
            action->set_param("src", node);
            action->set_param("dest", value_desc.get_value_node());

            assert(action->is_ready());

            if (!action->is_ready()) {
                throw Error(Error::TYPE_NOTREADY);
            }

            add_action_front(action);
        } else if (value_desc.parent_is_layer()) {
            Action::Handle action = LayerParamConnect::create();
            action->set_param("layer", value_desc.get_layer());
            action->set_param("param", value_desc.get_param_name());
            action->set_param("canvas", get_canvas());
            action->set_param("canvas_interface", get_canvas_interface());
            action->set_param("value_node", node);

            assert(action->is_ready());

            if (!action->is_ready()) {
                throw Error(Error::TYPE_NOTREADY);
            }

            add_action_front(action);
        } else if (value_desc.parent_is_value_node()) {
            Action::Handle action = ValueNodeLinkConnect::create();
            action->set_param("canvas", get_canvas());
            action->set_param("canvas_interface", get_canvas_interface());
            action->set_param("parent_value_node", value_desc.get_parent_value_node());
            action->set_param("index", value_desc.get_index());
            action->set_param("value_node", node);

            assert(action->is_ready());

            if (!action->is_ready()) {
                throw Error(Error::TYPE_NOTREADY);
            }

            add_action_front(action);
        }
    }
}